package com.hc.programming.tree;

import com.hc.programming.util.ArrayUtil;

import java.util.Arrays;

/**
 * 给你两棵二叉树： root1 和 root2 。
 * 想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。
 * 返回合并后的二叉树。
 * 注意: 合并过程必须从两个树的根节点开始。
 * <p>
 * 示例 1：
 * <a href="./合并二叉树-示例1.png">示例1</a>
 * 输入：root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
 * 输出：[3,4,5,5,4,null,7]
 * 示例 2：
 * 输入：root1 = [1], root2 = [1,2]
 * 输出：[2,2]
 * <p>
 * 提示：
 * 两棵树中的节点数目在范围 [0, 2000] 内
 * -10^4 <= Node.val <= 10^4
 *
 * @author huangchao E-mail:fengquan8866@163.com
 * @version 创建时间：2024/10/15 22:03
 */
public class 合并二叉树 {
    public static void main(String[] args) {
        Integer[] arr1 = new Integer[]{1, 3, 2, 5};
        Integer[] arr2  = new Integer[]{2, 1, 3, null, 4, null, 7};
        TreeNode tree1 = TreeNode.tree(arr1);
        TreeNode tree2 = TreeNode.tree(arr2);
        System.out.println(Arrays.toString(arr1) + "--" + Arrays.toString(arr2) + "=,--" + ArrayUtil.toString(mergeTrees(tree1, tree2)));
        arr1 = new Integer[]{1};
        arr2 = new Integer[]{1, 2};
        tree1 = TreeNode.tree(arr1);
        tree2 = TreeNode.tree(arr2);
        System.out.println(Arrays.toString(arr1) + "--" + Arrays.toString(arr2) + "=,--" + ArrayUtil.toString(mergeTrees(tree1, tree2)));
    }

    public static TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        if (root1 == null && root2 == null) return null;
        if (root1 == null) return root2;
        if (root2 == null) return root1;
        TreeNode root = new TreeNode(root1.val + root2.val);
        if (root1.left != null || root2.left != null) root.left = mergeTrees(root1.left, root2.left);
        if (root1.right != null || root2.right != null) root.right = mergeTrees(root1.right, root2.right);
        return root;
    }
}
